package com.itheima.reggie.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.itheima.reggie.common.CustomException;
import com.itheima.reggie.common.R;
import com.itheima.reggie.dto.SetmealDto;
import com.itheima.reggie.entity.Category;
import com.itheima.reggie.entity.Setmeal;
import com.itheima.reggie.entity.SetmealDish;
import com.itheima.reggie.service.CategoryService;
import com.itheima.reggie.service.SetmealDishService;
import com.itheima.reggie.service.SetmealService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cache.annotation.CacheEvict;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.*;

import java.util.List;
import java.util.stream.Collectors;

@RestController
@RequestMapping("/setmeal")
@Slf4j
public class SetmealController {

    @Autowired
    private SetmealService setmealService;

    @Autowired
    private CategoryService categoryService;

    @Autowired
    private SetmealDishService setmealDishService;

    /**
     * 保存套餐的方法
     * @param setmealDto 套餐的实体类
     * @return 返回添加套餐的结果
     */
    @PostMapping
    @CacheEvict(value = "setmealCache", allEntries = true) // 同时将该分类下的所有缓存数据都删除
    public R<String> save(@RequestBody SetmealDto setmealDto) {
        // 调用方法，保存套餐
        setmealService.saveWithDish(setmealDto);
        // 返回提示信息
        return R.success("套餐保存成功");
    }

    /**
     * 套餐分页
     * @param page 页码
     * @param pageSize 每页条数
     * @param name 模糊查询
     * @return 返回一页数据
     */
    @GetMapping("/page")
    public R<Page> page(int page, int pageSize, String name) {
        // 1.查询套餐分页对象
        Page<Setmeal> setmealPage = new Page<>(page, pageSize);
        // 声明 SetmealDto 对象
        Page<SetmealDto> setmealDtoPage = new Page<>();

        // 2.创建查询条件
        LambdaQueryWrapper<Setmeal> wrapper = new LambdaQueryWrapper<>();
        wrapper.like(StringUtils.isNotEmpty(name), Setmeal::getName, name);
        wrapper.orderByDesc(Setmeal::getCreateTime);

        // 3.进行分页查询
        setmealService.page(setmealPage, wrapper);

        // 4.将 setmealPage 的属性赋值给 setmealDtoPage（除了 records）
        BeanUtils.copyProperties(setmealPage, setmealDtoPage, "records");

        // 5.将查询出来的 records 更新，并放入到 setmealDtoPage 对象中
        List<Setmeal> records = setmealPage.getRecords();
        List<SetmealDto> list = records.stream().map((item) -> {
            SetmealDto setmealDto = new SetmealDto();
            // 属性复制
            BeanUtils.copyProperties(item, setmealDto);
            // 分类id
            Long categoryId = item.getCategoryId();
            // 根据分类id查找分类对象
            Category category = categoryService.getById(categoryId);
            // 赋值
            if (category != null) {
                String categoryName = category.getName();
                setmealDto.setCategoryName(categoryName);
            }
            return setmealDto;
        }).collect(Collectors.toList());

        // 将重新赋值的records赋值给setmealDtoPage
        setmealDtoPage.setRecords(list);

        // 返回结果
        return R.success(setmealDtoPage);
    }

    /**
     * 删除套餐
     * @param ids 套餐的id集合
     * @return 返回删除的结果信息
     */
    @DeleteMapping
    @Transactional
    @CacheEvict(value = "setmealCache", allEntries = true) // 同时将该分类下的所有缓存数据都删除
    public R<String> delete(@RequestParam List<Long> ids) {
        // 1.判断要删除的套餐是否有在销售的
        // 1-1 创建查询条件
        LambdaQueryWrapper<Setmeal> wrapper = new LambdaQueryWrapper<>();
        // 1-2 添加条件id
        wrapper.in(Setmeal::getId, ids);
        // 1-3 添加状态条件
        wrapper.eq(Setmeal::getStatus, 1);
        // 1-4 进行查询，查看有没有要删除的套餐
        int count = setmealService.count(wrapper);
        // 1-5 如果要删除正在销售的套餐，则不能进行删除，抛出异常
        if (count > 0) {
            throw new CustomException("正在启售的套餐不能删除");
        }

        // 2.如果全部套餐都是停售的，那么可以进行删除(setmeal表)
        setmealService.removeByIds(ids);

        // 3.当前套餐相关的菜品也要进行删除(setmeal_dish)
        // 3-1 创建查找条件
        LambdaQueryWrapper<SetmealDish> queryWrapper = new LambdaQueryWrapper<>();
        // 3-2 添加过滤条件
        queryWrapper.in(SetmealDish::getSetmealId, ids);
        // 3-3 执行删除语句
        setmealDishService.remove(queryWrapper);

        return R.success("数据删除成功");
    }

    /**
     * 通过套餐类别的id，查询这个类别下所有的套餐
     * @param setmeal
     * @return
     */
    @GetMapping("/list")
    @Cacheable(value = "setmealCache", key = "#setmeal.categoryId + '_' + #setmeal.status")
    public R<List<Setmeal>> list(Setmeal setmeal) {
        // 1.创建查询条件
        LambdaQueryWrapper<Setmeal> wrapper = new LambdaQueryWrapper<>();
        // 将这个类别下的所有套餐都查询出来
        wrapper.eq(setmeal.getCategoryId() != null ,Setmeal::getCategoryId, setmeal.getCategoryId());
        // 并且查询的是正在销售中的套餐
        wrapper.eq(setmeal.getStatus() != null, Setmeal::getStatus, 1);

        // 2.调用方法查询
        List<Setmeal> list = setmealService.list(wrapper);

        // 3.将查询的结果返回
        return R.success(list);
    }

}
